python3 train.py \
    --task=cifar10 \
    --dataset-root="/private/benchmarks/cifar10;./mask" \
    --ckpt-path=./cifar10_ckpt